package com.storage.utill;

import java.io.BufferedOutputStream;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;

public class FileUtil {
    /**
     *
     * @param filePath 文件路径
     * @return 返回文件的byte数组
     */
    public static byte[] getFileByte(String filePath){
        Path path = Paths.get(filePath);
        byte[] bytes = null;
        try {
            bytes= Files.readAllBytes(path);
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
        return bytes;
    }
    /**
     *
     * @param filePath 文件存储路径
     * @param fileByte 文件字节
     * @return 返回存储成功或者失败
     */
    public static boolean saveFile(String filePath, byte[] fileByte) {
        BufferedOutputStream bos = null;
        try {
            File file = new File(filePath);
            // 创建父目录（如果不存在）
            File parentDir = file.getParentFile();
            if (parentDir != null && !parentDir.exists()) {
                if (!parentDir.mkdirs()) {
                    return false;
                }
            }

            // 使用 FileOutputStream + BufferedOutputStream
            bos = new BufferedOutputStream(Files.newOutputStream(file.toPath()));

            // 分段写入，使用缓存块（4KB）
            int bufferSize = 4096;
            int offset = 0;
            while (offset < fileByte.length) {
                int length = Math.min(bufferSize, fileByte.length - offset);
                bos.write(fileByte, offset, length);
                offset += length;
            }

            bos.flush(); // 确保写入磁盘
            return true;
        } catch (IOException e) {
            e.printStackTrace();
            return false;
        } finally {
            if (bos != null) {
                try {
                    bos.close();
                } catch (IOException ignore) {}
            }
        }
    }

    public static InputStream getResourceInputStream(String resourceFileName){
        return FileUtil.class.getClassLoader().getResourceAsStream(resourceFileName);
    }

    public static void mkdirs(File file){
        if (!file.exists()) {
            file.mkdirs();
        }
    }
}
